import random

from django.http import HttpResponse, JsonResponse
from django.shortcuts import render

# Create your views here.
from django.views import View

from django_redis import get_redis_connection

from meiduo_pro.libs.captcha.captcha import captcha

# 返回图形验证码的类视图
from meiduo_pro.libs.yuntongxun.sms import CCP


class ImageCodeView(View):
    def get(self, request, uuid):
        # 1.调用工具类 captcha 生成图形验证码
        text, image = captcha.generate_captcha()

        # 2.链接 redis, 获取链接对象
        redis_conn = get_redis_connection('verify_code')

        # 3.利用链接对象, 保存数据到 redis, 使用 setex 函数
        redis_conn.setex('img_%s' % uuid, 300, text)
        print('图形验证码:', text)

        # 4.返回(图片)
        return HttpResponse(image,
                            content_type='image/jpg')

# 短信验证码后端接口
class SMSCodeView(View):
    def get(self, request, mobile):
        # 3. 创建连接到redis的对象 将这句话提到前面最开始的位置:
        redis_conn = get_redis_connection('verify_code')

########################################################################
        # 进入函数后, 先获取存储在 redis 中的数据
        send_flag = redis_conn.get('send_flag_%s' % mobile)
        if send_flag:
            return JsonResponse({'code': 400,
                                      'errmsg': '发送短信过于频繁'})
########################################################################

        # 1. 接收参数
        # image_code_client: 用户输入的短信验证码
        image_code_client = request.GET.get('image_code')
        uuid = request.GET.get('image_code_id')

        # 2. 校验参数
        if not all([image_code_client,uuid]):
            return JsonResponse({'code': 400,
                                      'errmsg': '缺少必传参数'})


        # 4. 提取图形验证码
        image_code_server = redis_conn.get('img_%s' % uuid)
        if image_code_server is None:
            # 图形验证码过期或者不存在
            return JsonResponse({'code': 400,
                                      'errmsg': '图形验证码失效'})
        # 5. 删除图形验证码，避免恶意测试图形验证码
        redis_conn.delete('img_%s' % uuid)

        # 6. 对比图形验证码
        # bytes 转字符串
        image_code_server = image_code_server.decode()
        # 转小写后比较
        if image_code_client.lower() != image_code_server.lower():
            return JsonResponse({'code': 400,
                                      'errmsg': '输入图形验证码有误'})

        # 7. 生成短信验证码：生成6位数验证码
        sms_code = '%06d' % random.randint(0, 999999)

        """创建管道对象:"""
        """
        可以一次性发送多条命令并在执行完后一次性将结果返回。
        pipeline 通过减少客户端与Redis的通信次数来实现降低往返延时时间。
        """
        pl = redis_conn.pipeline()
        # 8. 保存短信验证码
        # 短信验证码有效期，单位：300秒
        pl.setex('sms_%s' % mobile,
                         300,
                         sms_code)

########################################################################
        # 往 redis 中写入一个数据, 写入什么不重要, 时间重要
        # 我们给写入的数据设置为60s,如果过期,则会获取不到.
        pl.setex('send_flag_%s' % mobile, 60, 1)
########################################################################
        """执行管道: """
        pl.execute()

        # 9. 发送短信验证码
        # 注意： 测试的短信模板编号为1
        # ccp.send_template_sms('在官网添加的测试号码', ['验证码', 5], 1)

        # 原来的写法:
        # a = CCP().send_template_sms(mobile, [sms_code, 5], 1)
        # 改为现在的写法, 注意: 这里的函数,调用的时候需要加: .delay()
        from celery_tasks.sms.tasks import ccp_send_sms_code
        a = ccp_send_sms_code.delay(mobile, sms_code)


        print('短信验证码:', sms_code)
        print('发送状态:', a)
        # 10. 响应结果
        return JsonResponse({'code': 0,
                                  'errmsg': '发送短信成功'})